313C - Ilya and Matrix - CodeForces Solution

def beauty(a, size):
    if size == 0:
        return max(a)
    else:
        total = max(a) + sum(a)
        b = sorted(a, reverse=True)
        
        power = 1
        while (size-1) > 0:
            total += sum(b[:4 ** power])
            power += 1
            size -= 1
        return total

import math

n = int(input())
a = [int(i) for i in input().split()]
size = int(math.log(n, 4))

print(beauty(a, size))